/*
 * Copyright 2008 the Breachwalls OSS project and the Others.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
 * either express or implied. See the License for the specific language
 * governing permissions and limitations under the License.
 */
package com.breachwalls.mogen.generator;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

import org.dom4j.Document;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.breachwalls.mogen.generator.executor.FreeMarkerModelGenerationExecutor;
import com.breachwalls.mogen.generator.executor.ModelGenerationExecutor;
import com.breachwalls.mogen.model.Model;
import com.breachwalls.mogen.utils.Dom4jUtils;
import com.breachwalls.mogen.xml.context.DefaultModelParseContext;
import com.breachwalls.mogen.xml.context.ModelParseContext;
import com.breachwalls.mogen.xml.parser.DefaultModelGenerationDocumentParser;
import com.breachwalls.mogen.xml.parser.ModelGenerationDocumentParser;

/**
 * FreeMarkerを使用したデータタイプのモデルジェネレータ.
 * @author todoken
 * @version $Revision: 31 $
 */
public class FreeMarkerModelGenerator implements ModelGenerator {

    /**
     * ロガー.
     */
    private static Logger logger = LoggerFactory.getLogger(FreeMarkerModelGenerator.class);

    /**
     * モデル自動生成の外部ファイル設定.
     */
    private ModelGenerationProperties modelGenerationProperties;

    /**
     * モデル自動生成の実行役.
     */
    private ModelGenerationExecutor modelGenerationExecutor;

    /**
     * コンストラクタ.
     */
    public FreeMarkerModelGenerator() {
        modelGenerationProperties = new LocalModelGenerationProperties();
        modelGenerationExecutor = new FreeMarkerModelGenerationExecutor(modelGenerationProperties);
    }

    /**
     * コンストラクタ.
     * @param properties モデル自動生成の外部ファイル設定
     */
    public FreeMarkerModelGenerator(ModelGenerationProperties properties) {
        modelGenerationProperties = properties;
        modelGenerationExecutor = new FreeMarkerModelGenerationExecutor(modelGenerationProperties);
    }

    /**
     * モデル自動生成の外部ファイル設定を取得する.
     * @return モデル自動生成の外部ファイル設定
     */
    public ModelGenerationProperties getModelGenerationProperties() {
        return modelGenerationProperties;
    }

    /**
     * モデル自動生成の実行役を取得する.
     * @return モデル自動生成の実行役
     */
    public ModelGenerationExecutor getModelGenerationExecutor() {
        return modelGenerationExecutor;
    }

    /**
     * モデル自動生成の実行役を設定する.
     * @param newModelGenerationExecutor モデル自動生成の実行役
     */
    public void setModelGenerationExecutor(ModelGenerationExecutor newModelGenerationExecutor) {
        this.modelGenerationExecutor = newModelGenerationExecutor;
    }

    /**
     * モデル自動生成の定義（XML）ファイルを検索する.
     * @return 自動生成定義が書かれたドキュメント
     */
    protected Document[] findDocuments() {
        List<Document> documents = new ArrayList<Document>();
        File[] dirs = getModelGenerationProperties().getConfigDirectories();
        for (File dir : dirs) {
            FileFinder finder = new FileFinder(dir);
            Document[] docs = Dom4jUtils.createDocument(finder.find());
            for (Document doc : docs) {
                documents.add(doc);
            }
        }
        return documents.toArray(new Document[documents.size()]);
    }

    /**
     * モデル自動生成定義が書かれたドキュメントをパースし、モデルコンテキストに保存する.
     * @param documents モデル自動生成定義が書かれたドキュメント
     * @return 作成したモデルコンテキスト
     */
    protected ModelParseContext parseDocuments(Document[] documents) {
        ModelParseContext context = new DefaultModelParseContext();
        ModelGenerationDocumentParser parser = new DefaultModelGenerationDocumentParser();
        for (Document doc : documents) {
            parser.parse(doc, context);
        }
        context.compile();
        return context;
    }

    /**
     * {@inheritDoc}
     */
    public void generate() throws ModelGenerationException {
        Document[] documents = findDocuments();
        ModelParseContext context = parseDocuments(documents);
        generate(context);
    }

    /**
     * モデルコンテキストに保存されている内容をもとに、モデルの自動生成を行う.
     * @param context モデルコンテキスト
     * @throws ModelGenerationException モデルの自動生成に失敗したときに発生する例外
     */
    protected void generate(ModelParseContext context) throws ModelGenerationException {
        for (Model model : context.getAllModels()) {
            getModelGenerationExecutor().execute(model);
        }
    }

    /**
     * {@inheritDoc}
     */
    public static void main(final String[] args) {
        ModelGenerator generator = new FreeMarkerModelGenerator();
        try {
            generator.generate();
            if (logger.isInfoEnabled()) {
                logger.info("model generation successful.");
            }
            System.exit(0);
        } catch (ModelGenerationException e) {
            if (logger.isErrorEnabled()) {
                logger.error("model generation failure.", e);
            }
            System.exit(1);
        }
    }

}
